package com.linzm.leetcode.primary.exercises4_20230129;

/**
 * @Author zimingl
 * @Date 2023/2/9 23:19
 * @Description: 矩阵对角线元素的和
 * 给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。
 * 请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
 */
public class Demo16_1572 {
    public static void main(String[] args) {

    }

    private int diagonalSum(int[][] mat) {
        int sum = 0;
        for (int i = 0; i < mat.length; i++) {
            for (int j = 0; j < mat[i].length; j++) {
                if (i == j || (i + j) == (mat.length - 1)) {
                    sum += mat[i][j];
                }
            }
        }
        return sum;
    }
}
